from knnModelBased import *
# from test_windw import *
import sys,os

def main():
    # read data
    data = np.genfromtxt('/root/IdeaProjects/finish/word2vec/predict.txt',  delimiter=" ", skip_header=False)

    x = data[:, 1:100]
    y = data[:, 0]
    index = int(sys.argv[1])
    
    classes, y = np.unique(y, return_inverse=True)

    get_result(x, y, 10, index)

    # kfold_cross_validation(x, y, 10)
    
    
if __name__ == "__main__":
    main()
